# !/usr/bin/env python
# -*- coding: utf-8 -*-
# @File  : 调用chatAI模型.py
# @Author: dongguangwen
# @Date  : 2025-01-05 21:06
from langchain.chat_models import ChatOpenAI

chat = ChatOpenAI(model="Pro-128K",
                  temperature=0.8,
                  max_tokens=60,
                  api_key='dzCDkUHufCZEsuWlfVQd:wNCwEPgQSpfpARemq',
                  base_url='https://spark-api-open.xf-yun.com/v1'
                  )
from langchain.schema import (
    HumanMessage,
    SystemMessage
)

messages = [
    SystemMessage(content="你是一个成语接龙高手,进行成语接龙"),
    HumanMessage(content="好高骛远")
]
response = chat(messages)
print(response)

"""
content='我接一个“**远走高飞**”。\n\n它是指像鸿鹄一样飞向远方，形容人或事物远离原来的环境，追求更广阔的发展空间。这个成语的出处可以追溯到战国时期的《孟子·尽心上》。在《孟子·尽心上》中，孟子'
"""